/**
 *	4.修改前一个问题中的identity_matrix函数，它可以对数组进行扩展，从而能够接受任意大小的矩阵参数。
 *	函数的第一个参数应该是一个整型指针，你需要第二个参数，用于指定矩阵的大小。
**/

#include <stdio.h>

int identity_matrix(int *matrix, int n)
{
	int row, col;
	for(row = 0; row < n; row++)
		for(col = 0; col < n; col++)
			if(*matrix++ != (row == col))
				return 0; 
	return 1;
}

int main()
{
	int arr[][3] = {
		{1, 0, 0},
		{0, 1, 0},
		{0, 0, 1}
	};

	printf("%d\n", identity_matrix((int *)arr, 3));

	return 0;
}
